from os import path

import torch

import DQN

model = DQN.DQN()

if path.exists('./dqn.net'):
    model.load_state_dict(torch.load('./dqn.net'))

model.cpu()

torch.save(model.state_dict(), './dqn.cpu.net')
